from libs.StockData import StockData
from case.draw_kline import *
from case.test_kdj import *
from case.test_macd import *
from case.test_sma import *
from case.test_open_buy import *
from case.test_find_pattern import *
from case.test_trend import *
from case.test_bottom_high_volume import *
from case.test_zscore import *
from case.test_liner_fit import *

code_list = ['sh.000001', 'sz.399303', 'sh.000300', 'sz.399001', 'sh.000852']

if __name__ == '__main__':
    # test_draw_kline(code='sz.000001', start_date='2022-01-01', end_date='2025-02-28', frequency='d')
    # test_kdj(code='sz.399303', start_date='2017-01-01', end_date='2025-03-1', frequency='d')
    # test_macd(code='sh.000001', start_date='2017-01-01', end_date='2025-02-28', frequency='d')
    # test_sma(code='sz.002818', start_date='2020-01-01', end_date='2025-04-14', frequency='d')
    # for code in code_list:
    #     test_open_buy(code=code, start_date='2024-10-30', end_date='2025-03-09', frequency='d', duty=0.000, commision=0.0001)
    # for code in code_list:
    #     test_find_thursday_pattern(code=code, start_date='2024-10-30', end_date='2025-03-09', frequency='d')
    # test_trend(code='sh.603208', start_date='2019-8-30', end_date='2025-03-16', frequency='d', test=1)
    # test_trend_operation(code='sh.600702', start_date='2019-8-30', end_date='2024-09-16', frequency='d', text="下降趋势反转")
    # test_bottom_high_volume(code='sh.603501', start_date='2018-9-30', end_date='2025-3-17', frequency='d')
    # test_zscore(code='sz.002818', start_date='2018-9-30', end_date='2025-3-17', frequency='d')
    test_liner_fit(code='sz.002818', start_date='2025-01-01', end_date='2025-05-06', frequency='d')